<?php

namespace ezswoole;

use SplFileObject;

class File extends SplFileObject {
	/**
	 * 错误信息
	 * @var string
	 */
	private $error = '';
	// 当前完整文件名
	protected $filename;
	// 上传文件名
	protected $saveName;
	// 文件上传命名规则
	protected $rule = 'date';
	// 文件上传验证规则
	protected $validate = [];
	// 单元测试
	protected $isTest;
	// 上传文件信息
	protected $info;
	// 文件hash信息
	protected $hash = [];

	public function __construct($filename, $mode = 'r') {
		parent::__construct($filename, $mode);
		$this->filename = $this->getRealPath() ?: $this->getPathname();
	}

	/**
	 * 是否测试
	 * @param  bool   $test 是否测试
	 * @return $this
	 */
	public function isTest($test = false) {
		$this->isTest = $test;
		return $this;
	}

	/**
	 * 设置上传信息
	 * @param  array   $info 上传文件信息
	 * @return $this
	 */
	public function setUploadInfo($info) {
		$this->info = $info;
		return $this;
	}

	/**
	 * 获取上传文件的信息
	 * @param  string   $name
	 * @return array|string
	 */
	public function getInfo($name = '') {
		return isset($this->info[$name]) ? $this->info[$name] : $this->info;
	}

	/**
	 * 获取上传文件的文件名
	 * @return string
	 */
	public function getSaveName() {
		return $this->saveName;
	}

	/**
	 * 设置上传文件的保存文件名
	 * @param  string   $saveName
	 * @return $this
	 */
	public function setSaveName($saveName) {
		$this->saveName = $saveName;
		return $this;
	}

	/**
	 * 获取文件的哈希散列值
	 * @param string $type
	 * @return mixed $string
	 */
	public function hash($type = 'sha1') {
		if (!isset($this->hash[$type])) {
			$this->hash[$type] = hash_file($type, $this->filename);
		}
		return $this->hash[$type];
	}

	/**
	 * 检查目录是否可写
	 * @param  string   $path    目录
	 * @return boolean
	 */
	protected function checkPath($path) {
		if (is_dir($path)) {
			return true;
		}

		if (mkdir($path, 0755, true)) {
			return true;
		} else {
			$this->error = "目录 {$path} 创建失败！";
			return false;
		}
	}

	/**
	 * 获取文件类型信息
	 * @return string
	 */
	public function getMime() {
		$finfo = finfo_open(FILEINFO_MIME_TYPE);
		return finfo_file($finfo, $this->filename);
	}

	/**
	 * 设置文件的命名规则
	 * @param  string   $rule    文件命名规则
	 * @return $this
	 */
	public function rule($rule) {
		$this->rule = $rule;
		return $this;
	}

	/**
	 * 设置上传文件的验证规则
	 * @param  array   $rule    验证规则
	 * @return $this
	 */
	public function validate($rule = []) {
		$this->validate = $rule;
		return $this;
	}

	/**
	 * 检测是否合法的上传文件
	 * @return bool
	 */
	public function isValid() {
		if ($this->isTest) {
			return is_file($this->filename);
		}
		return is_uploaded_file($this->filename);
	}

	/**
	 * 检测上传文件
	 * @param  array   $rule    验证规则
	 * @return bool
	 */
	public function check($rule = []) {
		$rule = $rule ?: $this->validate;

		/* 检查文件大小 */
		if (isset($rule['size']) && !$this->checkSize($rule['size'])) {
			$this->error = '上传文件大小不符！';
			return false;
		}

		/* 检查文件Mime类型 */
		if (isset($rule['type']) && !$this->checkMime($rule['type'])) {
			$this->error = '上传文件MIME类型不允许！';
			return false;
		}

		/* 检查文件后缀 */
		if (isset($rule['ext']) && !$this->checkExt($rule['ext'])) {
			$this->error = '上传文件后缀不允许';
			return false;
		}

		/* 检查图像文件 */
		if (!$this->checkImg()) {
			$this->error = '非法图像文件！';
			return false;
		}

		return true;
	}

	/**
	 * 检测上传文件后缀
	 * @param  array|string   $ext    允许后缀
	 * @return bool
	 */
	public function checkExt($ext) {
		if (is_string($ext)) {
			$ext = explode(',', $ext);
		}
		$extension = strtolower(pathinfo($this->getInfo('name'), PATHINFO_EXTENSION));
		if (!in_array($extension, $ext)) {
			return false;
		}
		return true;
	}

	/**
	 * 检测图像文件
	 * @return bool
	 */
	public function checkImg() {
		$extension = strtolower(pathinfo($this->getInfo('name'), PATHINFO_EXTENSION));
		/* 对图像文件进行严格检测 */
		if (in_array($extension, ['gif', 'jpg', 'jpeg', 'bmp', 'png', 'swf']) && !in_array($this->getImageType($this->filename), [1, 2, 3, 4, 6, 13])) {
			return false;
		}
		return true;
	}

	// 判断图像类型
	protected function getImageType($image) {
		if (function_exists('exif_imagetype')) {
			return exif_imagetype($image);
		} else {
			$info = getimagesize($image);
			return $info[2];
		}
	}

	/**
	 * 检测上传文件大小
	 * @param  integer   $size    最大大小
	 * @return bool
	 */
	public function checkSize($size) {
		if ($this->getSize() > $size) {
			return false;
		}
		return true;
	}

	/**
	 * 检测上传文件类型
	 * @param  array|string   $mime    允许类型
	 * @return bool
	 */
	public function checkMime($mime) {
		if (is_string($mime)) {
			$mime = explode(',', $mime);
		}
		if (!in_array(strtolower($this->getMime()), $mime)) {
			return false;
		}
		return true;
	}

	/**
	 * 移动文件
	 * @param  string           $path    保存路径
	 * @param  string|bool      $savename    保存的文件名 默认自动生成
	 * @param  boolean          $replace 同名文件是否覆盖
	 * @return false|File false-失败 否则返回File实例
	 */
	public function move($path, $savename = true, $replace = true) {
		// 文件上传失败，捕获错误代码
		if (!empty($this->info['error'])) {
			$this->error($this->info['error']);
			return false;
		}

		// 检测合法性
		if (!$this->isValid()) {
			$this->error = '非法上传文件';
			return false;
		}

		// 验证上传
		if (!$this->check()) {
			return false;
		}
		$path = rtrim($path, DS) . DS;
		// 文件保存命名规则
		$saveName = $this->buildSaveName($savename);
		$filename = $path . $saveName;

		// 检测目录
		if (false === $this->checkPath(dirname($filename))) {
			return false;
		}

		/* 不覆盖同名文件 */
		if (!$replace && is_file($filename)) {
			$this->error = '存在同名文件' . $filename;
			return false;
		}

		/* 移动文件 */
		if ($this->isTest) {
			rename($this->filename, $filename);
		} elseif (!move_uploaded_file($this->filename, $filename)) {
			$this->error = '文件上传保存错误！';
			return false;
		}
		// 返回 File对象实例
		$file = new self($filename);
		$file->setSaveName($saveName);
		$file->setUploadInfo($this->info);
		return $file;
	}

	/**
	 * 获取保存文件名
	 * @param  string|bool   $savename    保存的文件名 默认自动生成
	 * @return string
	 */
	protected function buildSaveName($savename) {
		if (true === $savename) {
			// 自动生成文件名
			if ($this->rule instanceof \Closure) {
				$savename = call_user_func_array($this->rule, [$this]);
			} else {
				switch ($this->rule) {
				case 'date':
					$savename = date('Ymd') . DS . md5(microtime(true));
					break;
				default:
					if (in_array($this->rule, hash_algos())) {
						$hash     = $this->hash($this->rule);
						$savename = substr($hash, 0, 2) . DS . substr($hash, 2);
					} elseif (is_callable($this->rule)) {
						$savename = call_user_func($this->rule);
					} else {
						$savename = date('Ymd') . DS . md5(microtime(true));
					}
				}
			}
		} elseif ('' === $savename) {
			$savename = $this->getInfo('name');
		}
		if (!strpos($savename, '.')) {
			$savename .= '.' . pathinfo($this->getInfo('name'), PATHINFO_EXTENSION);
		}
		return $savename;
	}

	/**
	 * 获取错误代码信息
	 * @param int $errorNo  错误号
	 */
	private function error($errorNo) {
		switch ($errorNo) {
		case 1:
		case 2:
			$this->error = '上传文件大小超过了最大值！';
			break;
		case 3:
			$this->error = '文件只有部分被上传！';
			break;
		case 4:
			$this->error = '没有文件被上传！';
			break;
		case 6:
			$this->error = '找不到临时文件夹！';
			break;
		case 7:
			$this->error = '文件写入失败！';
			break;
		default:
			$this->error = '未知上传错误！';
		}
	}

	/**
	 * 获取错误信息
	 * @return mixed
	 */
	public function getError() {
		return $this->error;
	}

	public function __call($method, $args) {
		return $this->hash($method);
	}
}
